how to apply logic in pandas